import torch
from torch import nn
from torch.nn import ReLU

input = torch.tensor([[1,-0.5],
                     [-1,3]])

input = torch.reshape(input,(-1,1,2,2))

class MyModule(nn.Module):
    def __init__(self):
        super(MyModule,self).__init__()
        self.relu1=ReLU()

    def forward(self,input):
        output = self.relu1(input)
        return output
    
MyModule = MyModule()
output=MyModule(input)
print(output)
        



